package com.example.demo.剑指offer;

/**
 * @author：zzc
 * @date: 2022/8/19
 */
public class 树的子结构 {

    //输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
    //
    //B是A的子结构， 即 A中有出现和B相同的结构和节点值。
    //
    //例如:
    //给定的树 A:
    //
    //     3
    //
    //    / \
    //
    //   4   5
    //
    //  / \
    //
    // 1   2
    //给定的树 B：
    //
    //   4
    //
    //  /
    //
    // 1
    //返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

    //示例 1：
    //
    //输入：A = [1,2,3], B = [3,1]
    //输出：false
    //示例 2：
    //
    //输入：A = [3,4,5,1,2], B = [4,1]
    //输出：true

    /**
     * Definition for a binary tree node.
     * public class TreeNode {
     * int val;
     * TreeNode left;
     * TreeNode right;
     * TreeNode(int x) { val = x; }
     * }
     */
    public class TreeNode {

        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

    class Solution {

        public boolean isSubStructure(TreeNode A, TreeNode B) {
            if (A == null || B == null) {
                return false;
            }
            return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
        }

        private boolean dfs(TreeNode A, TreeNode B) {
            if (B == null) {
                return true;
            }
            if (A == null || A.val != B.val) {
                return false;
            }
            return dfs(A.left, B.left) && dfs(A.right, B.right);
        }
    }

}
